export const createWebSocket = <T>(url: string, dataHandler: (arg: T) => void): WebSocket => {
    const ws = new WebSocket(url);

    // 当连接打开时执行
    ws.onopen = function () {
        console.log("WebSocket 连接已打开");
        // 向服务器发送一个消息
        // ws.send("你好，服务器！");
    };

    // 当接收到服务器消息时执行
    ws.onmessage = function (event) {
        // console.log("收到服务器消息: " + event.data);
        const data = event.data
        dataHandler(JSON.parse(data))
    };

    // 当连接关闭时执行
    ws.onclose = function () {
        console.log("WebSocket 连接已关闭");
    };

    // 当连接出错时执行
    ws.onerror = function (error) {
        console.log("WebSocket 发生错误: ", error);
    };
    return ws
}
